import os

USER_HOME = os.path.expanduser("~")
BASE_DIR = os.path.join(USER_HOME, ".sg27")
EXCEL_DIR = os.path.join(BASE_DIR, "excel")
SQL_DIR = os.path.join(BASE_DIR, "sql")
CONFIG_PATH = os.path.join(BASE_DIR, "config.txt")

def initialize_environment():
    """创建必要目录结构"""
    os.makedirs(BASE_DIR, exist_ok=True)
    os.makedirs(EXCEL_DIR, exist_ok=True)
    os.makedirs(SQL_DIR, exist_ok=True)
    
    # 创建默认配置文件（如果不存在）
    if not os.path.exists(CONFIG_PATH):
        create_default_config()
    
    print(f"初始化完成: 目录结构已创建在 {BASE_DIR}")
    print(f"请将Excel文件放入: {EXCEL_DIR}")
    print(f"配置文件位置: {CONFIG_PATH}")

def create_default_config():
    """创建默认配置文件"""
    import configparser
    
    config = configparser.ConfigParser()
    config["database"] = {
        "host": "localhost",
        "user": "your_username",
        "password": "your_password",
        "database": "your_database",
        "batch_size": "50"
    }
    
    with open(CONFIG_PATH, "w", encoding="utf-8") as configfile:
        config.write(configfile)
    
    print(f"已创建默认配置文件: {CONFIG_PATH}")
    print("请修改配置文件中的数据库连接信息")